package org.msfj.common.util;

import cn.hutool.core.date.DateUtil;
import org.msfj.common.enums.Gender;

import java.text.ParseException;
import java.text.SimpleDateFormat;
import java.util.Calendar;
import java.util.Date;
import java.util.GregorianCalendar;
import java.util.Hashtable;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

public class IdCardUtil {

    public static boolean validate(String idNum) {
        if (idNum == null) {
            return false;
        }
        String num = "";

        if (idNum.length() == 18) {
            num = idNum.substring(0, 17);
        } else if (idNum.length() == 15) {
            num = idNum.substring(0, 6) + "19" + idNum.substring(6, 15);
        } else {
            return false;
        }

        if (!isNumeric(num)) {
            // "身份证15位号码都应为数字 ; 18位号码除最后一位外，都应为数字。"
            return false;
        }

        // 判断出生年月是否有效
        String strYear = num.substring(6, 10);
        String strMonth = num.substring(10, 12);
        String strDay = num.substring(12, 14);
        if (!isDate(strYear + "-" + strMonth + "-" + strDay)) {
            // "身份证出生日期无效。"
            return false;
        }

        GregorianCalendar gc = new GregorianCalendar();
        SimpleDateFormat s = new SimpleDateFormat("yyyy-MM-dd");
        try {
            if ((gc.get(Calendar.YEAR) - Integer.parseInt(strYear)) > 150
                    || (gc.getTime().getTime() - s.parse(strYear + "-" + strMonth + "-" + strDay).getTime()) < 0) {
                // "身份证生日不在有效范围。"
                return false;
            }
        } catch (ParseException e) {
            e.printStackTrace();
        }
        if (Integer.parseInt(strMonth) > 12 || Integer.parseInt(strMonth) == 0) {
            // "身份证月份无效"
            return false;
        }
        if (Integer.parseInt(strDay) > 31 || Integer.parseInt(strDay) == 0) {
            // "身份证日期无效"
            return false;
        }

        Hashtable areaCode = GetAreaCode();
        if (areaCode.get(num.substring(0, 2)) == null) {
            // "身份证地区编码错误。";
            return false;
        }

        return isVerifyCode(num, idNum);
    }

    /**
     * 从身份证中获取出生日期
     *
     * @param idNum
     * @return
     */
    public static Date getBirthday(String idNum) throws ParseException {
        String substring = idNum.substring(6, 14);
        Date birthday = DateUtil.parse(substring, "yyyyMMdd");
        return birthday;
    }

    /**
     * 从身份证中获取性别
     *
     * @param idNum
     * @return
     */
    public static Integer getGender(String idNum) {
        if (Integer.parseInt(idNum.substring(16, 17)) % 2 == 1) {
            return Gender.M.getCode();
        } else if (Integer.parseInt(idNum.substring(16, 17)) % 2 == 0) {
            return Gender.F.getCode();
        } else {
            return Gender.U.getCode();
        }
    }

    /*
     * 判断第18位校验码是否正确
     * 第18位校验码的计算方式：
       　　1. 对前17位数字本体码加权求和
           　　公式为：S = Sum(num * Wi), i = 0, ... , 16
           　　其中num表示第i个位置上的身份证号码数字值，Wi表示第i位置上的加权因子，其各位对应的值依次为： 7 9 10 5 8 4 2 1 6 3 7 9 10 5 8 4 2
       　　2. 用11对计算结果取模
       　　   Y = mod(S, 11)
       　　3. 根据模的值得到对应的校验码
           　　对应关系为：
           　　Y值：    0  1  2  3  4  5  6  7  8  9  10
           　　校验码： 1  0  X  9  8  7  6  5  4  3   2
     */
    private static boolean isVerifyCode(String num, String idNum) {
        String[] VerifyCode = { "1", "0", "X", "9", "8", "7", "6", "5", "4","3", "2" };
        String[] Wi = { "7", "9", "10", "5", "8", "4", "2", "1", "6", "3", "7","9", "10", "5", "8", "4", "2" };
        int sum = 0;
        for (int i = 0; i < 17; i++) {
            sum = sum + Integer.parseInt(String.valueOf(num.charAt(i))) * Integer.parseInt(Wi[i]);
        }
        int modValue = sum % 11;
        String strVerifyCode = VerifyCode[modValue];
        num = num + strVerifyCode;
        if (idNum.length() == 18 && !num.equals(idNum)) {
            return false;
        }
        return true;
    }


    /**
     * 将所有地址编码保存在一个Hashtable中
     * @return Hashtable 对象
     */
    private static Hashtable GetAreaCode() {
        Hashtable hashtable = new Hashtable();
        hashtable.put("11", "北京");
        hashtable.put("12", "天津");
        hashtable.put("13", "河北");
        hashtable.put("14", "山西");
        hashtable.put("15", "内蒙古");
        hashtable.put("21", "辽宁");
        hashtable.put("22", "吉林");
        hashtable.put("23", "黑龙江");
        hashtable.put("31", "上海");
        hashtable.put("32", "江苏");
        hashtable.put("33", "浙江");
        hashtable.put("34", "安徽");
        hashtable.put("35", "福建");
        hashtable.put("36", "江西");
        hashtable.put("37", "山东");
        hashtable.put("41", "河南");
        hashtable.put("42", "湖北");
        hashtable.put("43", "湖南");
        hashtable.put("44", "广东");
        hashtable.put("45", "广西");
        hashtable.put("46", "海南");
        hashtable.put("50", "重庆");
        hashtable.put("51", "四川");
        hashtable.put("52", "贵州");
        hashtable.put("53", "云南");
        hashtable.put("54", "西藏");
        hashtable.put("61", "陕西");
        hashtable.put("62", "甘肃");
        hashtable.put("63", "青海");
        hashtable.put("64", "宁夏");
        hashtable.put("65", "新疆");
        hashtable.put("71", "台湾");
        hashtable.put("81", "香港");
        hashtable.put("82", "澳门");
        hashtable.put("91", "国外");
        return hashtable;
    }

    /**
     * 判断字符串是否为数字,0-9重复0次或者多次
     * @param str
     * @return
     */
    private static boolean isNumeric(String str) {
        Pattern pattern = Pattern.compile("[0-9]*");
        Matcher isNum = pattern.matcher(str);
        return isNum.matches();
    }

    /**
     * 功能：判断字符串出生日期是否符合正则表达式：包括年月日，闰年、平年和每月31天、30天和闰月的28天或者29天
     *
     * @param str
     * @return
     */
    private static boolean isDate(String str) {
        Pattern pattern = Pattern
                .compile("^((\\d{2}(([02468][048])|([13579][26]))[\\-\\/\\s]?((((0?[13578])|(1[02]))[\\-\\/\\s]?((0?[1-9])|([1-2][0-9])|(3[01])))|(((0?[469])|(11))[\\-\\/\\s]?((0?[1-9])|([1-2][0-9])|(30)))|(0?2[\\-\\/\\s]?((0?[1-9])|([1-2][0-9])))))|(\\d{2}(([02468][1235679])|([13579][01345789]))[\\-\\/\\s]?((((0?[13578])|(1[02]))[\\-\\/\\s]?((0?[1-9])|([1-2][0-9])|(3[01])))|(((0?[469])|(11))[\\-\\/\\s]?((0?[1-9])|([1-2][0-9])|(30)))|(0?2[\\-\\/\\s]?((0?[1-9])|(1[0-9])|(2[0-8]))))))?$");
        Matcher isDate = pattern.matcher(str);
        return isDate.matches();
    }
}